package 数组;

/**
 * @author: 大怪
 * @email: 962527441@qq.com
 */

import java.util.HashMap;

/**
 * 题目：
 * 给定一个二进制数组 nums , 找到含有相同数量的 0 和 1 的最长连续子数组，并返回该子数组的长度
 *
 * 示例 1:
 *
 * 输入: nums = [0,1]
 * 输出: 2
 * 说明: [0, 1] 是具有相同数量 0 和 1 的最长连续子数组
 */
public class 子数组中0和1个数相同的 {
    class Solution {
        public int findMaxLength(int[] nums) {
            HashMap<Integer, Integer> map = new HashMap<Integer, Integer>();
            int len = 0;
            int sum = 0;
            map.put(0, -1);
            for (int i = 0; i < nums.length; i++) {
                sum += nums[i]==0 ? -1:1;
                if (map.containsKey(sum)) {
                    len = Math.max(len, i-map.get(sum));
                }else {
                    map.put(sum, i);
                }
            }
            return len;
        }
    }
}
